Skip to content

Migrate pynvml to cuda.core.system#146

Open
mdboom wants to merge 1 commit into
rapidsai:mainfrom
mdboom:pynvml-to-cuda.core.system
Open

Migrate pynvml to cuda.core.system#146
mdboom wants to merge 1 commit into
rapidsai:mainfrom
mdboom:pynvml-to-cuda.core.system

Conversation

@mdboom
Copy link
Copy Markdown

@mdboom mdboom commented Apr 28, 2026

Migrates from pynvml.py to the new Cython/cybind-based cuda.core.system API.

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Apr 28, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@mdboom mdboom force-pushed the pynvml-to-cuda.core.system branch from f6a44df to d88499b Compare May 19, 2026 15:47
@mdboom mdboom marked this pull request as ready for review May 19, 2026 15:47
@mdboom mdboom requested review from a team as code owners May 19, 2026 15:47
@mdboom mdboom requested a review from KyleFromNVIDIA May 19, 2026 15:47
@ncclementi
Copy link
Copy Markdown
Contributor

ncclementi commented May 21, 2026

I was trying to install this in Colab for example to give it a try and see if it works as expected, and I ran into an installation ERROR

ERROR: pip's dependency resolver does not currently take into account all the 
packages that are installed. This behaviour is the source of the following dependency conflicts.
torch 2.10.0+cu128 requires cuda-bindings==12.9.4; platform_system == "Linux", but you 
have cuda-bindings 13.2.0 which is incompatible.
cuda-python 12.9.4 requires cuda-bindings~=12.9.4, but you have cuda-bindings 13.2.0 which is incompatible.
numba-cuda 0.22.2 requires cuda-core<1.0.0,>=0.3.2, but you have cuda-core 1.0.1 which is incompatible.
dask-cuda 26.2.0 requires cuda-core==0.3.*, but you have cuda-core 1.0.1 which is incompatible.

It seems to be updating cuda bindings to the latest, which is to aggressive for most CSP.

I understand that the numba cuda version in colab is super old. So we can report that back and see if we can get them to upgrade.

All that being said, we can run the script. At the moment we are running the script to check the environment, but it seems like with this changes installing the rapids-cli is modifying that environment.

I'd like to know what @jacobtomlinson, @jayavenkatesh19 and @mmccarty think here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants